#include "widget.h"
#include "ui_widget.h"

Widget::Widget(QWidget *parent)
    : QWidget(parent)
    , ui(new Ui::Widget)
{
    ui->setupUi(this);

    // 设置节点的名字
    ui->treeWidget->setHeaderLabel("动物");

    // 新增顶层节点
    QTreeWidgetItem* item1 = new QTreeWidgetItem();
    // 每个节点都可以设置多个列，此处为了简单就设置一列
    item1->setText(0, "猫");
    // 添加到顶层节点中
    ui->treeWidget->addTopLevelItem(item1);

    // 新增顶层节点
    QTreeWidgetItem* item2 = new QTreeWidgetItem();
    // 每个节点都可以设置多个列，此处为了简单就设置一列
    item2->setText(0, "狗");
    // 添加到顶层节点中
    ui->treeWidget->addTopLevelItem(item2);

    // 新增顶层节点
    QTreeWidgetItem* item3 = new QTreeWidgetItem();
    // 每个节点都可以设置多个列，此处为了简单就设置一列
    item3->setText(0, "鸟");
    // 添加到顶层节点中
    ui->treeWidget->addTopLevelItem(item3);

    // 添加一些子节点
    QTreeWidgetItem* item4 = new QTreeWidgetItem();
    item4->setText(0, "中华田园猫");
    item1->addChild(item4);

    QTreeWidgetItem* item5 = new QTreeWidgetItem();
    item5->setText(0, "布偶猫");
    item1->addChild(item5);

    QTreeWidgetItem* item6 = new QTreeWidgetItem();
    item6->setText(0, "暹罗猫");
    item1->addChild(item6);
}

Widget::~Widget()
{
    delete ui;
}


void Widget::on_pushButton_insertTop_clicked()
{
    // 获取到输入框的内容
    const QString& text = ui->lineEdit->text();
    // 构造一个QTreeWidgetItem
    QTreeWidgetItem* item = new QTreeWidgetItem();
    item->setText(0, text);
    // 添加到顶层节点中
    ui->treeWidget->addTopLevelItem(item);
}

void Widget::on_pushButton_insert_clicked()
{
    // 获取到当前选中的节点
    QTreeWidgetItem* curitem = ui->treeWidget->currentItem();
    if(curitem == nullptr){
        return;
    }
    // 获取到输入框的内容
    const QString& text = ui->lineEdit->text();
    // 构造一个QTreeWidgetItem
    QTreeWidgetItem* item = new QTreeWidgetItem();
    item->setText(0, text);
    // 插入到选中节点的子节点中
    curitem->addChild(item);
}

void Widget::on_pushButton_del_clicked()
{
    // 获取到当前选中的节点
    QTreeWidgetItem* curitem = ui->treeWidget->currentItem();
    if(curitem == nullptr){
        return;
    }
    // 删除选中的元素，需要先获取父节点，通过父节点进行删除
    QTreeWidgetItem* parent = curitem->parent();
    if(parent == nullptr){
        // 顶层元素
        int index = ui->treeWidget->indexOfTopLevelItem(curitem);
        ui->treeWidget->takeTopLevelItem(index);
    }else{
        // 普通元素
        parent->removeChild(curitem);
    }

}
